package com.hjq.utils;

import com.aliyun.oss.*;
import com.aliyun.oss.common.auth.CredentialsProviderFactory;
import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider;
import com.aliyun.oss.common.comm.SignVersion;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import java.io.ByteArrayInputStream;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.UUID;

/**
 * 阿里云对象存储服务(OSS)操作工具类
 * 
 * 该类提供了上传文件到阿里云OSS的功能，支持自动日期目录管理和文件名去重。
 * 
 * 使用前请确保已设置环境变量：
 * - OSS_ACCESS_KEY_ID：阿里云访问密钥ID
 * - OSS_ACCESS_KEY_SECRET：阿里云访问密钥Secret
 * 
 * @author hjq
 * @version 1.0
 */
@Component
public class AliyunOSSOperator {

    // /**
    //  * OSS服务接入点
    //  */
    // @Value("${aliyun.oss.endpoint}")
    // private String endpoint ;
    
    // /**
    //  * 存储空间名称
    //  */
    // @Value("${aliyun.oss.bucketName}")
    // private String bucketName;
    
    // /**
    //  * 地域标识
    //  */
    // @Value("${aliyun.oss.region}")
    // private String region ;

    /**
     * 上传文件到阿里云OSS
     * 
     * 该方法会自动创建以当前年月为结构的目录（如：2024/06），
     * 并为上传的文件生成唯一的文件名以避免冲突。
     * 
     * @param content 文件的字节数组内容
     * @param originalFilename 原始文件名，用于获取文件扩展名
     * @return 上传文件的完整访问URL
     * @throws Exception 上传过程中可能抛出的异常，如网络错误、认证失败等
     */
    @Autowired
    private AliyunOSSProperties aliyunOSSProperties;
    public String upload(byte[] content, String originalFilename) throws Exception {
        String endpoint=aliyunOSSProperties.getEndpoint();
        String bucketName=aliyunOSSProperties.getBucketName();
        String region=aliyunOSSProperties.getRegion();
        // 从环境变量中获取访问凭证。运行本代码示例之前，请确保已设置环境变量OSS_ACCESS_KEY_ID和OSS_ACCESS_KEY_SECRET。
        EnvironmentVariableCredentialsProvider credentialsProvider = CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();

        // 填写Object完整路径，例如202406/1.png。Object完整路径中不能包含Bucket名称。
        //获取当前系统日期的字符串,格式为 yyyy/MM
        String dir = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM"));
        //生成一个新的不重复的文件名
        String newFileName = UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
        String objectName = dir + "/" + newFileName;

        // 创建OSSClient实例。
        ClientBuilderConfiguration clientBuilderConfiguration = new ClientBuilderConfiguration();
        clientBuilderConfiguration.setSignatureVersion(SignVersion.V4);
        OSS ossClient = OSSClientBuilder.create()
                .endpoint(endpoint)
                .credentialsProvider(credentialsProvider)
                .clientConfiguration(clientBuilderConfiguration)
                .region(region)
                .build();

        try {
            ossClient.putObject(bucketName, objectName, new ByteArrayInputStream(content));
        } finally {
            ossClient.shutdown();
        }

        return endpoint.split("//")[0] + "//" + bucketName + "." + endpoint.split("//")[1] + "/" + objectName;
    }

}
